Searched refs:tensor_slices (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | mean_stddev_normalization.cc | 92 const int tensor_slices) in MeanStdDevNormalization() argument
100 std::min(tensor_slices, gpu_info.GetMaxWorkGroupSizeForX()); in MeanStdDevNormalization()
137 while (desired_work_group_size >= tensor_slices * 2) { in MeanStdDevNormalization()
219 const int tensor_slices) { in CreateMeanStdDevNormalization() argument
220 return MeanStdDevNormalization(definition, gpu_info, tensor_slices); in CreateMeanStdDevNormalization()
|
| D | mean_stddev_normalization.h | 32 const int tensor_slices);
55 const int tensor_slices);
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 115 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice() argument
116 DCHECK_NE(tensor_slices, nullptr); in RegisterTensorSlice()
117 TensorSliceSet* tss = gtl::FindPtrOrNull(*tensor_slices, name); in RegisterTensorSlice()
121 tensor_slices->insert(std::make_pair(name, tss)); in RegisterTensorSlice()
|
| D | tensor_slice_set.h | 87 std::unordered_map<string, TensorSliceSet*>* tensor_slices);
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 70 tensor_slices = []
80 tensor_slices.append(spec.slice_spec)
83 return io_ops.save_v2(file_prefix, tensor_names, tensor_slices, tensors)
105 tensor_names, tensor_slices, tensor_dtypes = zip(*restore_specs)
109 file_prefix, tensor_names, tensor_slices, tensor_dtypes)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
68 if tensor_slices is None:
71 return gen_io_ops.save_slices(filename, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 168 std::vector<TensorSlice> tensor_slices; in Compute() local
184 tensor_slices.push_back(slice); in Compute()
202 for (const TensorSlice& tensor_slice : tensor_slices) { in Compute()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 105 tensor_slices = []
110 tensor_slices.append(spec.slice_spec)
116 tensor_slices=tensor_slices)
120 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
|