Searched refs:shape_output (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 176 Tensor shape_output; in TryToInferTensorOutputFromStridedSliceNode() local
178 *shape_node, shape_c, &shape_output, success, /*allow_partial=*/true)); in TryToInferTensorOutputFromStridedSliceNode()
189 if (evaluated && node.output_type(0) == shape_output.dtype()) { in TryToInferTensorOutputFromStridedSliceNode()
194 if (shape_output.dtype() == DT_INT32 && in TryToInferTensorOutputFromStridedSliceNode()
195 shape_output.flat<int>()(begin_value) != -1) { in TryToInferTensorOutputFromStridedSliceNode()
196 t.flat<int32>()(0) = shape_output.flat<int>()(begin_value); in TryToInferTensorOutputFromStridedSliceNode()
199 } else if (shape_output.dtype() == DT_INT64 && in TryToInferTensorOutputFromStridedSliceNode()
200 shape_output.flat<int64_t>()(begin_value) != -1) { in TryToInferTensorOutputFromStridedSliceNode()
201 t.flat<int64_t>()(0) = shape_output.flat<int64_t>()(begin_value); in TryToInferTensorOutputFromStridedSliceNode()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GEMMLowpFixture.h | 95 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs…
104 TensorType output = create_tensor<TensorType>(shape_output, data_type_output, 1);
126 …alse, false, reshape_b_only_on_first_run, (reinterpret_output_as_3d ? shape_output[2] : 0), reinte…
174 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs…
221 …transpose_A ? a_transposed : a), (pretranspose_B ? b_transposed : b), shape_output, a_offset, b_of…
226 …transpose_A ? a_transposed : a), (pretranspose_B ? b_transposed : b), shape_output, a_offset, b_of…
235 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup() argument
237 _target = compute_target(shape_a, shape_b, shape_output, a_offset, b_offset); in setup()
238 _reference = compute_reference(shape_a, shape_b, shape_output, a_offset, b_offset); in setup()
242 … TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_output, int32_t a_offs… in compute_target() argument
[all …]
|