| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 86 const auto& actual_shape = argument_buffers[i].on_device_shape(); in ExecuteAsyncOnStream() local
89 if (!ShapeUtil::DynamicArrayShapeIsCompatible(actual_shape, in ExecuteAsyncOnStream()
94 actual_shape)) { in ExecuteAsyncOnStream()
101 ShapeUtil::HumanStringWithLayout(actual_shape)); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 164 auto match_shape = [](const TensorShape& actual_shape, in AreShapesCompatible()
167 if (actual_shape.dims() != cached_shape.dims()) return false; in AreShapesCompatible()
170 if (actual_shape.dim_size(0) > cached_shape.dim_size(0)) return false; in AreShapesCompatible()
172 for (int i = 1; i < actual_shape.dims(); ++i) { in AreShapesCompatible()
173 if (actual_shape.dim_size(i) != cached_shape.dim_size(i)) return false; in AreShapesCompatible()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stack_ops.cc | 80 TensorShape actual_shape; in MaybeInitializeStack() local
81 TF_RETURN_IF_ERROR(GetStackShape(builder, resource, &actual_shape)); in MaybeInitializeStack()
82 if (stack_shape != actual_shape) { in MaybeInitializeStack()
85 actual_shape.DebugString()); in MaybeInitializeStack()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | einsum_op_test.py | 223 for actual_shape, placeholder_shape in input_and_placeholder_shapes:
224 with self.subTest(equation=equation, actual_shape=actual_shape,
226 input_np = np.array(r.randn(*actual_shape))
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 148 TensorShape actual_shape = val.shape(); in SetRetval() local
149 actual_shape.InsertDim(0, compute_opts_->batch_size); in SetRetval()
151 compute_opts_->output.allocate(index, actual_shape, &out)); in SetRetval()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 340 const Shape& actual_shape = arguments[i].Buffers().shape(); in ExecuteAsyncOnStream() local
342 ShapeUtil::DynamicShapeIsCompatible(actual_shape, expected_shape)) in ExecuteAsyncOnStream()
345 << actual_shape.ToString(/*print_layout=*/true); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | convert_to_constants_test.py | 89 actual_shape = input_data[tensor.name.split(":")[0]].shape
90 self.assertEqual(tensor.shape, actual_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 430 TensorShapeProto actual_shape = result.first->second.shape(); in SetShapeAttribute() local
433 if (actual_shape.ShortDebugString() != tshape.ShortDebugString()) { in SetShapeAttribute()
436 actual_shape.ShortDebugString()); in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2052 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties() local
2053 if (!InferenceContext::RankKnown(actual_shape)) { in AsTensorProperties()
2056 for (int j = 0; j < InferenceContext::Rank(actual_shape); ++j) { in AsTensorProperties()
2058 InferenceContext::DimKnownRank(actual_shape, j); in AsTensorProperties()
2067 const auto& actual_shape = shapes_.GetMergedValue(s); in GetMergedShape() local
2068 if (!InferenceContext::RankKnown(actual_shape)) { in GetMergedShape()
2072 for (int j = 0; j < InferenceContext::Rank(actual_shape); ++j) { in GetMergedShape()
2074 InferenceContext::DimKnownRank(actual_shape, j); in GetMergedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 76 Shape actual_shape = b->GetShape(operand).ConsumeValueOrDie(); in CheckShape() local
77 CHECK(ShapeUtil::Equal(expected_shape, actual_shape)) in CheckShape()
79 << ShapeUtil::HumanString(actual_shape); in CheckShape()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 193 actual_shape = input_data[tensor.name.split(":")[0]].shape
194 self.assertEqual(tensor.shape, actual_shape)
|
| D | function_test.py | 563 actual_val, actual_shape = Foo()
568 self.assertAllEqual(expected_shape, self.evaluate(actual_shape))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops_test.py | 792 for actual_shape, placeholder_shape in input_and_placeholder_shapes:
793 input_np = np.array(r.randn(*actual_shape))
|