| /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 | layout_util.cc | 40 xla::Shape* xla_shape) { in RewriteLayoutWithShardedShape()
|
| D | xla_helpers.cc | 140 xla::Shape xla_shape; in IdentityShapeRepresentationFn() local
|
| D | xla_compiler_test.cc | 311 xla::Shape xla_shape; in TEST_F() local
357 xla::Shape xla_shape; in TEST_F() local
412 xla::Shape xla_shape; in TEST_F() local
1173 xla::Shape xla_shape; in TEST_F() local
1216 xla::Shape xla_shape; in TEST_F() local
|
| D | xla_op_kernel.cc | 578 xla::Shape xla_shape; in ReadVariableInputTensor() local
725 xla::Shape xla_shape; in AssignVariableTensor() local
|
| D | xla_compiler.cc | 770 xla::Shape xla_shape = std::get<xla::Shape>(args[i].shape); in CompileFunction() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 53 xla::Shape xla_shape; in Compile() local
87 xla::Shape xla_shape; in Compile() local
128 xla::Shape xla_shape; in Compile() local
159 xla::Shape xla_shape; in Compile() local
218 xla::Shape xla_shape; in Compile() local
|
| D | stateless_random_ops.cc | 163 xla::Shape xla_shape; in Compile() local
218 xla::Shape xla_shape; in Compile() local
259 xla::Shape xla_shape; in Compile() local
298 xla::Shape xla_shape; in Compile() local
352 xla::Shape xla_shape; in Compile() local
397 xla::Shape xla_shape; in Compile() local
|
| D | stateful_random_ops.cc | 228 xla::Shape xla_shape; in Compile() local
272 xla::Shape xla_shape; in Compile() local
312 xla::Shape xla_shape; in Compile() local
357 xla::Shape xla_shape; in Compile() local
390 xla::Shape xla_shape; in Compile() local
|
| D | stateless_random_ops_v2.cc | 237 xla::Shape xla_shape; in Compile() local
319 xla::Shape xla_shape; in Compile() local
369 xla::Shape xla_shape; in Compile() local
421 xla::Shape xla_shape; in Compile() local
492 xla::Shape xla_shape; in Compile() local
|
| D | strided_slice_op.cc | 312 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()
163 xla::Shape xla_shape; in Compile() local
439 xla::Shape xla_shape; in Compile() local
497 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 | infeed_ops.cc | 77 const xla::Shape& xla_shape) { in TransposeTensor()
307 xla::Shape xla_shape; in PrelinearizeTupleOp() local
480 xla::Shape xla_shape; in TpuInfeedEnqueueTupleOp() local
|
| D | tpu_reshard_variables_op_util.cc | 132 const xla::Shape& xla_shape = xla_tensor->shaped_buffer().on_host_shape(); in BuildInputBuffers() local
252 const xla::Shape& xla_shape = in UpdateOutputVariables() local
|
| D | outfeed_ops.h | 97 xla::Shape xla_shape; in TpuOutfeedDequeueTupleOp() local
|
| D | tpu_execute_op.cc | 230 const xla::Shape& xla_shape = xla_tensor->shaped_buffer().on_host_shape(); in BuildComputationInputs() local
420 const xla::Shape& xla_shape = in AllocateOutputTensors() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | layout_util.cc | 25 xla::Shape* xla_shape) { in RewriteLayoutWithShardedShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 134 xla::Shape& xla_shape = individual_arg_shapes.back(); in GetXlaInputShapes() local
192 const xla::Shape& xla_shape) -> StatusOr<xla::Shape> { in GetOutputInfo()
528 [&](const xla::Shape& xla_shape) -> StatusOr<mlir::XlaLayoutPreference> { in ConvertMLIRToXlaComputation()
538 mlir::XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in ConvertMLIRToXlaComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_conversions.cc | 233 void ToC(const xla::Shape& xla_shape, XLA_Shape* c_shape) { in ToC()
347 XLA_ShapeIndex ToC(const xla::ShapeIndex& xla_shape) { in ToC()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 59 xla::Shape xla_shape; in TpuShapeRepresentation() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 279 xla::Shape xla_shape; in XRTAllocateFromTensorOp() local
|