| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test.py | 47 def __getitem__(self, slice_spec): argument
48 return slice_spec
54 def _make_tensor_slice_spec(slice_spec, use_constant=True): argument
83 if isinstance(slice_spec, tuple):
84 return tuple(make_piece_scalar(piece) for piece in slice_spec)
86 return make_piece_scalar(slice_spec)
721 def _TestGetItem(self, rt, slice_spec, expected): argument
739 tensor_slice_spec1 = _make_tensor_slice_spec(slice_spec, True)
740 tensor_slice_spec2 = _make_tensor_slice_spec(slice_spec, False)
741 value1 = self.eval_to_list(rt.__getitem__(slice_spec))
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 42 slice_spec = ""
47 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
49 saver.BaseSaverBuilder.SaveSpec(num_updates, slice_spec,
51 saver.BaseSaverBuilder.SaveSpec(partition_ids, slice_spec,
53 saver.BaseSaverBuilder.SaveSpec(feature_ids, slice_spec,
55 saver.BaseSaverBuilder.SaveSpec(gradients, slice_spec,
57 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
|
| D | model_ops.py | 58 slice_spec = ""
60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
|
| D | quantile_ops.py | 52 slice_spec = ""
54 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object.py | 24 def __init__(self, tensor, slice_spec, name, dtype=None): argument
35 self.slice_spec = slice_spec
|
| D | saveable_object_util.py | 61 def __init__(self, var, slice_spec, name): argument
62 spec = saveable_object.SaveSpec(var, slice_spec, name, dtype=var.dtype)
79 def __init__(self, var, slice_spec, name): argument
103 spec = saveable_object.SaveSpec(tensor, slice_spec, name,
|
| D | functional_saver.py | 93 tensor_slices.append(spec.slice_spec)
126 restore_specs.append((spec.name, spec.slice_spec, spec.dtype))
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 332 bool IsFullSlice(const TensorSlice& slice_spec, in IsFullSlice() argument
334 if (slice_spec.IsFull()) { in IsFullSlice()
338 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
451 const TensorSlice& slice_spec, in AddSlice() argument
457 if (IsFullSlice(slice_spec, full_tensor_shape)) { in AddSlice()
480 slice_spec.AsProto(slice_proto); in AddSlice()
485 checkpoint::EncodeTensorNameSlice(full_tensor_key_string, slice_spec); in AddSlice()
924 const TensorSlice& slice_spec, Tensor* val) { in LookupSlice() argument
928 return GetSliceValue(full_tensor_key, entry, slice_spec, val); in LookupSlice()
933 const TensorSlice& slice_spec, Tensor* val) { in GetSliceValue() argument
[all …]
|
| D | tensor_bundle.h | 141 const TensorSlice& slice_spec, const Tensor& slice_tensor);
243 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
282 const TensorSlice& slice_spec,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 73 slice_spec = ''
77 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
175 slice_spec = ''
177 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
179 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 | 526 def _slice_helper(tensor, slice_spec, var=None): argument
586 if isinstance(slice_spec, bool) or \
587 (isinstance(slice_spec, ops.Tensor) and slice_spec.dtype == dtypes.bool) or \
588 (isinstance(slice_spec, np.ndarray) and slice_spec.dtype == bool):
589 return boolean_mask(tensor=tensor, mask=slice_spec)
591 if not isinstance(slice_spec, (list, tuple)):
592 slice_spec = [slice_spec]
600 for s in slice_spec:
865 def _SliceHelperVar(var, slice_spec): argument
908 return _slice_helper(var.value(), slice_spec, var)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| D | stats_ops.py | 67 slice_spec = ""
68 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
|
| D | model_ops.py | 70 slice_spec = ""
71 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | checkpoint_utils.py | 107 def _set_checkpoint_initializer(variable, file_pattern, tensor_name, slice_spec, argument
120 file_pattern, [tensor_name], [slice_spec], [base_type], name=name)[0]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 292 slice_spec, argument
313 ckpt_file, [tensor_name], [slice_spec], [base_type], name=name)[0]
|
| D | saver.py | 112 tensor_slices.append(spec.slice_spec)
180 [spec.slice_spec],
574 restore_specs.append((spec.name, spec.slice_spec, spec.dtype))
|
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| D | split_dependency.py | 34 slice_spec="",
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 709 slice_spec = ""
720 saver.BaseSaverBuilder.SaveSpec(param, slice_spec, param_name)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values.py | 1207 slice_spec="",
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_test.py | 220 slice_spec="",
|