Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc86 const auto& actual_shape = argument_buffers[i].on_device_shape(); in ExecuteAsyncOnStream() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dstack_ops.cc80 TensorShape actual_shape; in MaybeInitializeStack() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dutils.cc122 const TensorShape& cached_shape) { in AreShapesCompatible()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc333 const Shape& actual_shape = arguments[i].Buffers().shape(); in ExecuteAsyncOnStream() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_defun_op.cc150 TensorShape actual_shape = val.shape(); in SetRetval() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dexport_utils.cc453 TensorShapeProto actual_shape = result.first->second.shape(); in SetShapeAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_attributes.cc290 TensorShapeProto actual_shape = result.first->second.shape(); in SetShapeAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc2096 ShapeHandle actual_shape = shapes_.GetMergedValue(shape); in AsTensorProperties() local
2111 const auto& actual_shape = shapes_.GetMergedValue(s); in GetMergedShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc71 Shape actual_shape = b->GetShape(operand).value(); in CheckShape() local