Home
last modified time | relevance | path

Searched defs:xla_shape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dliteral_util.cc27 xla::Shape xla_shape; in HostTensorToBorrowingLiteral() local
43 xla::Shape xla_shape; in HostTensorToMutableBorrowingLiteral() local
50 const xla::Shape& xla_shape, Tensor* host_tensor, in HostTensorToMutableBorrowingLiteral()
Dxla_compiler.cc322 const xla::Shape& xla_shape = resource->representation_shape().value(); in BuildComputation() local
544 xla::Shape xla_shape; in XlaCompiler() local
709 xla::Shape xla_shape = absl::get<xla::Shape>(args[i].shape); in CompileFunction() local
769 xla::Shape* xla_shape) { in XLAShapeForArgument()
Dxla_op_kernel.cc420 xla::Shape xla_shape; in ReadVariableInputTensor() local
558 xla::Shape xla_shape; in AssignVariableTensor() local
Dxla_compiler_test.cc310 xla::Shape xla_shape; in TEST_F() local
354 xla::Shape xla_shape; in TEST_F() local
407 xla::Shape xla_shape; in TEST_F() local
1091 xla::Shape xla_shape; in TEST_F() local
1130 xla::Shape xla_shape; in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drandom_ops.cc48 xla::Shape xla_shape; in Compile() local
225 xla::Shape xla_shape; in Compile() local
264 xla::Shape xla_shape; in Compile() local
292 xla::Shape xla_shape; in Compile() local
325 xla::Shape xla_shape; in Compile() local
Dstateless_random_ops.cc124 xla::Shape xla_shape; in Compile() local
179 xla::Shape xla_shape; in Compile() local
218 xla::Shape xla_shape; in Compile() local
273 xla::Shape xla_shape; in Compile() local
Dstateful_random_ops.cc209 xla::Shape xla_shape; in Compile() local
253 xla::Shape xla_shape; in Compile() local
293 xla::Shape xla_shape; in Compile() local
338 xla::Shape xla_shape; in Compile() local
371 xla::Shape xla_shape; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util_test.cc34 xla::Shape xla_shape; in TestShapeRepresentation() local
Dcompile_mlir_util.cc78 xla::Shape& xla_shape = individual_arg_shapes.back(); in GetXlaInputShapes() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc97 xla::Shape xla_shape; in XlaDeviceContext() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h278 xla::Shape xla_shape; in XRTAllocateFromTensorOp() local