| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_getitem_test.py | 39 def __getitem__(self, slice_spec): argument
46 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
121 def _TestGetItem(self, rt, slice_spec, expected, expected_shape=None): argument
154 def _TestGetItemException(self, rt, slice_spec, expected, message): argument
245 def testWithRaggedRank1(self, slice_spec, expected): argument
307 def testErrorsWithRaggedRank1(self, slice_spec, expected, message): argument
388 def testWithRaggedRank2(self, slice_spec, expected): argument
414 def testErrorsWithRaggedRank2(self, slice_spec, expected, message): argument
427 def testWithEmptyTensor(self, slice_spec, expected): argument
438 def testErrorsWithEmptyTensor(self, slice_spec, expected, message): argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_slice_test.py | 36 def __getitem__(self, slice_spec): argument
45 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
152 def _TestGetItem(self, struct, slice_spec, expected): argument
205 def testGetitemFromScalarStruct(self, slice_spec, expected): argument
230 def testGetitemFromVectorStruct(self, slice_spec, expected, argument
253 def testGetItemError(self, slice_spec, error, exception=ValueError): argument
262 def testGetItemFromVectorError(self, slice_spec, error, exception=ValueError): argument
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | save_restore_util.cc | 66 std::string slice_spec = absl::StrJoin(slice_specs, ":"); in BuildSliceSpecDeviceMap() local
121 for (const auto& slice_spec : slice_specs) { in BuildSavingSpec() local
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object.py | 21 def __init__(self, tensor, slice_spec, name, dtype=None, device=None): argument
|
| D | saveable_object_util.py | 72 def __init__(self, var, slice_spec, name): argument
90 def __init__(self, var, slice_spec, name): argument
|
| D | saveable_object_util_test.py | 35 def __init__(self, var, slice_spec, name): argument
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 358 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice()
480 const TensorSlice& slice_spec, in AddSlice()
997 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice()
1006 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue()
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | d_variable.py | 39 slice_spec, argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 459 slice_spec, argument
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1509 def _slice_helper(tensor, slice_spec, update_method=None, updates=None): argument
1778 def _as_spec_tuple(slice_spec): argument
1795 def _getitem(self, slice_spec): argument
1810 def _with_index_update_helper(update_method, a, slice_spec, updates): argument
|
| D | np_utils.py | 662 def getitem(a, slice_spec): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 343 def __getitem__(self, slice_spec): argument
409 def _decompose_slice_spec(self, slice_spec): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 915 def _slice_helper(tensor, slice_spec, var=None): argument
1307 def _SliceHelperVar(var, slice_spec): argument
|