Home
last modified time | relevance | path

Searched defs:actual_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc86 const auto& actual_shape = argument_buffers[i].on_device_shape(); in ExecuteAsyncOnStream() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc80 TensorShape actual_shape; in MaybeInitializeStack() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc165 const TensorShape& cached_shape) { in AreShapesCompatible()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc148 TensorShape actual_shape = val.shape(); in SetRetval() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc340 const Shape& actual_shape = arguments[i].Buffers().shape(); in ExecuteAsyncOnStream() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc430 TensorShapeProto actual_shape = result.first->second.shape(); in SetShapeAttribute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2052 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties() local
2067 const auto& actual_shape = shapes_.GetMergedValue(s); in GetMergedShape() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc76 Shape actual_shape = b->GetShape(operand).ConsumeValueOrDie(); in CheckShape() local