Searched refs:tensor_type (Results 1 - 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| H A D | mpi_message.proto | 37 DataType tensor_type = 3;
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| H A D | types.h | 27 static ArrayDataType Deserialize(int tensor_type);
|
| H A D | types.cc | 67 ArrayDataType DataType::Deserialize(int tensor_type) { argument
68 switch (::tflite::TensorType(tensor_type)) {
80 LOG(FATAL) << "Unhandled tensor type '" << tensor_type << "'.";
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | load_and_remap_matrix_op.cc | 137 DataType tensor_type; variable
140 old_tensor_name, &tensor_type, &tensor_shape));
141 OP_REQUIRES(context, tensor_type == DT_FLOAT,
144 DataTypeString(tensor_type), " instead of expected type ",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | array_ops_test.py | 450 def __init__(self, test, x, tensor_type=dtypes.int32, check_type_infer=True):
451 self.x_np = np.array(x).astype(tensor_type.as_numpy_dtype)
453 if tensor_type.is_complex:
456 self.x = constant_op.constant(self.x_np, dtype=tensor_type)
500 for tensor_type in STRIDED_SLICE_TYPES:
501 with self.test_session(use_gpu=not tensor_type.is_integer):
503 self, StridedSliceChecker.REF_TENSOR, tensor_type=tensor_type)
521 checker2 = StridedSliceChecker(self, 5, tensor_type=tensor_type)
[all...] |
| /external/tensorflow/tensorflow/python/client/ |
| H A D | session.py | 138 tensor_type,
164 tensor_type: The type for which you want to register a conversion function.
165 fetch_function: A callable that takes an object of type `tensor_type` and
168 returns an object of some value type that corresponds to `tensor_type`.
173 `tensor_type`, and feed_tensor must have type `tf.Tensor`. Each feed
177 feed when setting up a partial run, which takes a `tensor_type` type
181 if issubclass(conversion_function[0], tensor_type):
183 tensor_type)
185 _REGISTERED_EXPANSIONS.insert(0, (tensor_type, fetch_function, feed_function,
250 for tensor_type, fetch_f
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | cluster_function_library_runtime_test.cc | 148 key: "tensor_type"
189 key: "tensor_type"
|
| /external/tensorflow/tensorflow/python/ops/ |
| H A D | io_ops.py | 120 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type,
131 tensor_type: the type of the tensor to restore.
136 A tensor of type "tensor_type".
138 base_type = dtypes.as_dtype(tensor_type).base_dtype
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| H A D | mpi_ops.cc | 306 auto data_type = requests[0].tensor_type();
308 auto request_type = requests[i].tensor_type();
|
| /external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 140 def register_dense_tensor_like_type(tensor_type):
141 """EXPERIMENTAL: Registers `tensor_type` as implementing the tensor interface.
147 tensor_type: A type implementing the tensor interface.
150 TypeError: If `tensor_type` does not implement the tensor interface.
153 if not isinstance(tensor_type.name, property):
155 tensor_type.__name__)
158 tensor_type.__name__)
160 if not isinstance(tensor_type.dtype, property):
162 tensor_type.__name__)
165 tensor_type
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | virtual_scheduler_test.cc | 377 key: "tensor_type"
|
Completed in 351 milliseconds