Home
last modified time | relevance | path

Searched defs:slice_spec (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_getitem_test.py43 def __getitem__(self, slice_spec): argument
50 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
125 def _TestGetItem(self, rt, slice_spec, expected, expected_shape=None): argument
158 def _TestGetItemException(self, rt, slice_spec, expected, message): argument
249 def testWithRaggedRank1(self, slice_spec, expected): argument
311 def testErrorsWithRaggedRank1(self, slice_spec, expected, message): argument
392 def testWithRaggedRank2(self, slice_spec, expected): argument
418 def testErrorsWithRaggedRank2(self, slice_spec, expected, message): argument
431 def testWithEmptyTensor(self, slice_spec, expected): argument
442 def testErrorsWithEmptyTensor(self, slice_spec, expected, message): argument
[all …]
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_slice_test.py40 def __getitem__(self, slice_spec): argument
49 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
156 def _TestGetItem(self, struct, slice_spec, expected): argument
209 def testGetitemFromScalarStruct(self, slice_spec, expected): argument
234 def testGetitemFromVectorStruct(self, slice_spec, expected, argument
257 def testGetItemError(self, slice_spec, error, exception=ValueError): argument
266 def testGetItemFromVectorError(self, slice_spec, error, exception=ValueError): argument
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py73 def __init__(self, var, slice_spec, name): argument
91 def __init__(self, var, slice_spec, name): argument
Dsaveable_object.py24 def __init__(self, tensor, slice_spec, name, dtype=None, device=None): argument
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc352 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice()
478 const TensorSlice& slice_spec, in AddSlice()
984 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice()
993 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue()
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py406 slice_spec, argument
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py329 def __getitem__(self, slice_spec): argument
394 def _decompose_slice_spec(self, slice_spec): argument
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py1507 def _slice_helper(tensor, slice_spec, update_method=None, updates=None): argument
1776 def _as_spec_tuple(slice_spec): argument
1793 def _getitem(self, slice_spec): argument
1808 def _with_index_update_helper(update_method, a, slice_spec, updates): argument
Dnp_utils.py633 def getitem(a, slice_spec): argument
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py896 def _slice_helper(tensor, slice_spec, var=None): argument
1251 def _SliceHelperVar(var, slice_spec): argument