Home
last modified time | relevance | path

Searched refs:actual_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dtensor_util.py133 def _shape_tensor_compatible(expected_shape, actual_shape): argument
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): argument
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)
202 actual_shape = array_ops.shape(actual_tensor, name='actual')
206 is_shape = _is_shape(expected_shape, actual_tensor, actual_shape)
211 actual_shape
[all …]
Dtensor_util_test.py191 actual_shape = [2, 2]
192 tensor = constant_op.constant(value, shape=actual_shape)
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc74 const auto& actual_shape = arguments[i]->on_device_shape(); in ExecuteOnStream() local
75 if (!ShapeUtil::Equal(expected_shape, actual_shape)) { in ExecuteOnStream()
79 ShapeUtil::HumanString(actual_shape)); in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc80 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/core/kernels/data/
Dmap_defun_op.cc233 TensorShape actual_shape = val.shape(); in SetRetval() local
234 actual_shape.InsertDim(0, compute_opts_->batch_size); in SetRetval()
236 compute_opts_->output.allocate(index, actual_shape, &out)); in SetRetval()
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
Dclassifier_metrics_impl.py633 actual_shape = var.get_shape()
636 if actual_shape != expected_shape:
638 actual_shape, expected_shape))
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc75 Shape actual_shape = b->GetShape(operand).ConsumeValueOrDie(); in CheckShape() local
76 CHECK(ShapeUtil::Equal(expected_shape, actual_shape)) in CheckShape()
78 << ShapeUtil::HumanString(actual_shape); in CheckShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1740 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties() local
1741 if (!InferenceContext::RankKnown(actual_shape)) { in AsTensorProperties()
1744 for (int j = 0; j < InferenceContext::Rank(actual_shape); ++j) { in AsTensorProperties()
1746 InferenceContext::DimKnownRank(actual_shape, j); in AsTensorProperties()
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py560 actual_val, actual_shape = Foo()
565 self.assertAllEqual(expected_shape, self.evaluate(actual_shape))