| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_slice_test.py | 40 def __getitem__(self, slice_spec): argument
41 return slice_spec
49 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
78 if isinstance(slice_spec, tuple):
79 return tuple(make_piece_scalar(piece) for piece in slice_spec)
81 return make_piece_scalar(slice_spec)
156 def _TestGetItem(self, struct, slice_spec, expected): argument
175 tensor_slice_spec1 = _make_tensor_slice_spec(slice_spec, True)
176 tensor_slice_spec2 = _make_tensor_slice_spec(slice_spec, False)
177 value1 = struct.__getitem__(slice_spec)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_getitem_test.py | 43 def __getitem__(self, slice_spec): argument
44 return slice_spec
50 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
79 if isinstance(slice_spec, tuple):
80 return tuple(make_piece_scalar(piece) for piece in slice_spec)
82 return make_piece_scalar(slice_spec)
125 def _TestGetItem(self, rt, slice_spec, expected, expected_shape=None): argument
145 tensor_slice_spec1 = _make_tensor_slice_spec(slice_spec, True)
146 tensor_slice_spec2 = _make_tensor_slice_spec(slice_spec, False)
147 value1 = rt.__getitem__(slice_spec)
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 329 def __getitem__(self, slice_spec): argument
350 if (isinstance(slice_spec, bool) or (isinstance(slice_spec, ops.Tensor) and
351 slice_spec.dtype == dtypes.bool) or
352 (isinstance(slice_spec, np.ndarray) and slice_spec.dtype == bool)):
354 return array_ops.boolean_mask(tensor=tensor, mask=slice_spec)
356 if not isinstance(slice_spec, (list, tuple)):
357 slice_spec = (slice_spec,)
359 s = slice_spec[0]
365 all_dim_slice_spec = (first_dim_slice_specs[i],) + slice_spec[1:]
375 return array_ops.concat([var[slice_spec] for var in self._variables],
[all …]
|
| D | values_util.py | 77 slice_spec="",
109 slice_spec="",
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object.py | 24 def __init__(self, tensor, slice_spec, name, dtype=None, device=None): argument
38 self.slice_spec = slice_spec
|
| D | saveable_object_util.py | 73 def __init__(self, var, slice_spec, name): argument
74 spec = saveable_object.SaveSpec(var, slice_spec, name, dtype=var.dtype)
91 def __init__(self, var, slice_spec, name): argument
120 spec = saveable_object.SaveSpec(tensor, slice_spec, name,
409 "slice_spec": spec.slice_spec})
|
| D | functional_saver.py | 80 tensor_slices.append(spec.slice_spec)
104 restore_specs.append((spec.name, spec.slice_spec, spec.dtype))
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1507 def _slice_helper(tensor, slice_spec, update_method=None, updates=None): argument
1550 for index, s in enumerate(slice_spec):
1671 dim = index - len(slice_spec) + num_shrink
1776 def _as_spec_tuple(slice_spec): argument
1778 if isinstance(slice_spec,
1779 (list, tuple)) and not isinstance(slice_spec, np.ndarray):
1781 for s in slice_spec:
1789 return tuple(slice_spec)
1790 return (slice_spec,)
1793 def _getitem(self, slice_spec): argument
[all …]
|
| D | np_utils.py | 633 def getitem(a, slice_spec): argument
635 return _maybe_static(a)[slice_spec]
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 352 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice() argument
354 if (slice_spec.IsFull()) { in IsFullSlice()
358 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
478 const TensorSlice& slice_spec, in AddSlice() argument
484 if (IsFullSlice(slice_spec, full_tensor_shape)) { in AddSlice()
507 slice_spec.AsProto(slice_proto); in AddSlice()
512 checkpoint::EncodeTensorNameSlice(full_tensor_key_string, slice_spec); in AddSlice()
984 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice() argument
988 return GetSliceValue(full_tensor_key, entry, slice_spec, val); in LookupSlice()
993 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue() argument
[all …]
|
| D | tensor_bundle.h | 141 const TensorSlice& slice_spec, const Tensor& slice_tensor);
244 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
283 const TensorSlice& slice_spec,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 79 slice_spec = ''
83 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
181 slice_spec = ''
183 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
185 saver.BaseSaverBuilder.SaveSpec(serialized, slice_spec,
|
| D | tensor_forest_ops.py | 43 slice_spec = ''
44 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name)]
|
| D | array_ops.py | 896 def _slice_helper(tensor, slice_spec, var=None): argument
967 return tensor._numpy_style_getitem(slice_spec) # pylint: disable=protected-access
969 if isinstance(slice_spec, bool) or \
970 (isinstance(slice_spec, ops.Tensor) and slice_spec.dtype == dtypes.bool) or \
971 (isinstance(slice_spec, np.ndarray) and slice_spec.dtype == bool):
972 return boolean_mask(tensor=tensor, mask=slice_spec)
974 if not isinstance(slice_spec, (list, tuple)):
975 slice_spec = [slice_spec]
983 for s in slice_spec:
1251 def _SliceHelperVar(var, slice_spec): argument
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__operators__.pbtxt | 13 …argspec: "args=[\'tensor\', \'slice_spec\', \'var\'], varargs=None, keywords=None, defaults=[\'Non…
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 406 slice_spec, argument
427 ckpt_file, [tensor_name], [slice_spec], [base_type], name=name)[0]
|
| D | saver.py | 110 tensor_slices.append(spec.slice_spec)
175 io_ops.restore_v2(filename_tensor, [spec.name], [spec.slice_spec],
578 restore_specs.append((spec.name, spec.slice_spec, spec.dtype))
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | unstack_op_test.py | 60 slice_spec = tuple(
62 result.append(data.__getitem__(slice_spec))
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | saving.py | 39 slice_spec="",
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 105 slice_spec = ":".join(
107 shape_and_slice = full_shape_str + slice_spec
|
| D | util_test.py | 175 slice_spec="",
|