MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / make_piece_scalar

Function make_piece_scalar

tensorflow/python/ops/ragged/ragged_tensor_test.py:71–83  ·  view source on GitHub ↗
(piece)

Source from the content-addressed store, hash-verified

69 """
70
71 def make_piece_scalar(piece):
72 if isinstance(piece, int):
73 scalar = constant_op.constant(piece)
74 if use_constant:
75 return scalar
76 else:
77 return array_ops.placeholder_with_default(scalar, [])
78 elif isinstance(piece, slice):
79 return slice(
80 make_piece_scalar(piece.start), make_piece_scalar(piece.stop),
81 make_piece_scalar(piece.step))
82 else:
83 return piece
84
85 if isinstance(slice_spec, tuple):
86 return tuple(make_piece_scalar(piece) for piece in slice_spec)

Callers 1

_make_tensor_slice_specFunction · 0.85

Calls 2

sliceFunction · 0.50
constantMethod · 0.45

Tested by

no test coverage detected