| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 27 xla::Shape xla_shape; in HostTensorToBorrowingLiteral() local
33 Status HostTensorToBorrowingLiteral(const xla::Shape& xla_shape, in HostTensorToBorrowingLiteral()
54 xla::Shape xla_shape; in HostTensorToMutableBorrowingLiteral() local
61 const xla::Shape& xla_shape, Tensor* host_tensor, in HostTensorToMutableBorrowingLiteral()
|
| D | xla_helpers.cc | 132 xla::Shape xla_shape; in IdentityShapeRepresentationFn() local
142 xla::Shape* xla_shape) { in RewriteLayoutWithShardedShape()
|
| D | xla_op_kernel.cc | 497 xla::Shape xla_shape; in ReadVariableInputTensor() local
640 xla::Shape xla_shape; in AssignVariableTensor() local
|
| D | xla_compiler_test.cc | 309 xla::Shape xla_shape; in TEST_F() local
352 xla::Shape xla_shape; in TEST_F() local
405 xla::Shape xla_shape; in TEST_F() local
1101 xla::Shape xla_shape; in TEST_F() local
1142 xla::Shape xla_shape; in TEST_F() local
|
| D | xla_compiler.cc | 761 xla::Shape xla_shape = absl::get<xla::Shape>(args[i].shape); in CompileFunction() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 49 xla::Shape xla_shape; in Compile() local
226 xla::Shape xla_shape; in Compile() local
265 xla::Shape xla_shape; in Compile() local
293 xla::Shape xla_shape; in Compile() local
326 xla::Shape xla_shape; in Compile() local
|
| D | stateless_random_ops.cc | 162 xla::Shape xla_shape; in Compile() local
217 xla::Shape xla_shape; in Compile() local
258 xla::Shape xla_shape; in Compile() local
297 xla::Shape xla_shape; in Compile() local
351 xla::Shape xla_shape; in Compile() local
|
| D | stateful_random_ops.cc | 227 xla::Shape xla_shape; in Compile() local
271 xla::Shape xla_shape; in Compile() local
311 xla::Shape xla_shape; in Compile() local
356 xla::Shape xla_shape; in Compile() local
389 xla::Shape xla_shape; in Compile() local
|
| D | stateless_random_ops_v2.cc | 223 xla::Shape xla_shape; in Compile() local
289 xla::Shape xla_shape; in Compile() local
336 xla::Shape xla_shape; in Compile() local
384 xla::Shape xla_shape; in Compile() local
439 xla::Shape xla_shape; in Compile() local
|
| D | strided_slice_op.cc | 275 xla::Shape xla_shape = operand_shape_or.ValueOrDie(); in Compile() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | outfeed_ops.cc | 37 xla::Shape xla_shape; in Compile() local
67 xla::Shape xla_shape; in Compile() local
|
| D | host_compute_ops.cc | 52 xla::Shape* xla_shape) { in MakeXlaShapes()
151 xla::Shape xla_shape; in Compile() local
417 xla::Shape xla_shape; in Compile() local
469 xla::Shape xla_shape; in Compile() local
|
| D | infeed_op.cc | 119 xla::Shape xla_shape; in InfeedDequeueTupleOp() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | outfeed_ops.cc | 77 xla::Shape xla_shape; in TpuOutfeedDequeueTupleOp() local
|
| D | infeed_ops.cc | 77 const xla::Shape& xla_shape) { in TransposeTensor()
307 xla::Shape xla_shape; in PrelinearizeTupleOp() local
460 xla::Shape xla_shape; in TpuInfeedEnqueueTupleOp() local
|
| D | tpu_execute_op.cc | 231 const xla::Shape& xla_shape = xla_tensor->shaped_buffer().on_host_shape(); in BuildComputationInputs() local
421 const xla::Shape& xla_shape = in AllocateOutputTensors() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 212 void ToC(const xla::Shape& xla_shape, XLA_Shape* c_shape) { in ToC()
317 XLA_ShapeIndex ToC(const xla::ShapeIndex& xla_shape) { in ToC()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 99 xla::Shape xla_shape; in XlaDeviceContext() local
|
| D | xla_tpu_device.cc | 56 xla::Shape xla_shape; in TpuShapeRepresentation() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 278 xla::Shape xla_shape; in XRTAllocateFromTensorOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 100 xla::Shape& xla_shape = individual_arg_shapes.back(); in GetXlaInputShapes() local
|