| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| tensor_util.py | 133 def _shape_tensor_compatible(expected_shape, actual_shape):
134 """Returns whether actual_shape is compatible with expected_shape.
140 actual_shape: Shape of the tensor to test.
145 values=[expected_shape, actual_shape]) as scope:
149 math_ops.equal(expected_shape, actual_shape, 'equal'),
169 def _is_shape(expected_shape, actual_tensor, actual_shape=None):
177 actual_shape: Shape of actual_tensor, if we already have it.
183 if actual_shape is None:
184 actual_shape = array_ops.shape(actual_tensor, name='actual')
185 shape_equal = _shape_tensor_compatible(expected_shape, actual_shape)
[all...] |
| tensor_util_test.py | 191 actual_shape = [2, 2]
192 tensor = constant_op.constant(value, shape=actual_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| executable.cc | 74 const auto& actual_shape = arguments[i]->on_device_shape(); local
75 if (!ShapeUtil::Equal(expected_shape, actual_shape)) {
79 ShapeUtil::HumanString(actual_shape));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| stack_ops.cc | 80 TensorShape actual_shape; local
81 TF_RETURN_IF_ERROR(GetStackShape(builder, resource, &actual_shape));
82 if (stack_shape != actual_shape) {
85 actual_shape.DebugString());
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_defun_op.cc | 233 TensorShape actual_shape = val.shape(); variable
234 actual_shape.InsertDim(0, compute_opts_->batch_size);
236 compute_opts_->output.allocate(index, actual_shape, &out));
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| classifier_metrics_impl.py | 633 actual_shape = var.get_shape()
636 if actual_shape != expected_shape:
638 actual_shape, expected_shape))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| batch_normalization_test.cc | 75 Shape actual_shape = b->GetShape(operand).ConsumeValueOrDie(); local
76 CHECK(ShapeUtil::Equal(expected_shape, actual_shape))
78 << ShapeUtil::HumanString(actual_shape);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 1740 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); local
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| function_test.py | 560 actual_val, actual_shape = Foo()
565 self.assertAllEqual(expected_shape, self.evaluate(actual_shape))
[all...] |
