Searched refs:XLAShapeToTensorShape (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 99 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(lhs_xla_shape, &lhs_tf_shape)); in BroadcastOpsToSame()
100 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(rhs_xla_shape, &rhs_tf_shape)); in BroadcastOpsToSame()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 144 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(input_shape, &input_tensor_shape)); in ConvBackpropComputeDimensionsV2XlaShapes()
145 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(filter_shape, &filter_tensor_shape)); in ConvBackpropComputeDimensionsV2XlaShapes()
147 XLAShapeToTensorShape(out_backprop_shape, &out_backprop_tensor_shape)); in ConvBackpropComputeDimensionsV2XlaShapes()
408 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(filter_shape, &filter_tensor_shape)); in MakeXlaBackpropFilterConvOp()
409 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(input_shape, &input_tensor_shape)); in MakeXlaBackpropFilterConvOp()
410 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(output_shape, &output_tensor_shape)); in MakeXlaBackpropFilterConvOp()
|
| D | dynamic_stitch_op.cc | 69 ctx, XLAShapeToTensorShape(indices_input[0].shape(), &indices0_shape)); in Compile()
73 XLAShapeToTensorShape(indices_input[input_num].shape(), in Compile()
|
| D | stack_ops.cc | 48 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0), in GetStackShape()
|
| D | tensor_array_ops.cc | 77 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in MaybeInitializeTensorArray()
|
| D | tensor_list_ops.cc | 412 OP_REQUIRES_OK(ctx, XLAShapeToTensorShape(buffer_xla_shape, &buffer_shape)); in Compile()
|
| D | pooling_ops.cc | 347 OP_REQUIRES_OK(ctx, XLAShapeToTensorShape(status_or_shape.ValueOrDie(), in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.h | 33 Status XLAShapeToTensorShape(const xla::Shape& shape,
|
| D | xla_helpers.cc | 167 XLAShapeToTensorShape(per_device_xla_shape, &per_device_tensor_shape)); in RewriteLayoutWithShardedShape()
199 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(original_shape, &shape)); in ReshapeWithCorrectRepresentationAndSharding()
|
| D | literal_util.cc | 115 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(literal.shape(), &shape)); in LiteralToHostTensor()
|
| D | xla_expression.cc | 211 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(xla_shape, &shape)); in GetShape()
|
| D | shape_util.cc | 88 Status XLAShapeToTensorShape(const xla::Shape& shape, in XLAShapeToTensorShape() function
|
| D | xla_op_kernel.cc | 632 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in AssignVariableTensor()
|
| D | xla_compiler.cc | 765 if (XLAShapeToTensorShape(xla_shape, &tensor_shape).ok() && in CompileFunction()
847 XLAShapeToTensorShape(absl::get<xla::Shape>(arg.shape), &shape)); in XLAShapeForArgument()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 74 XLAShapeToTensorShape(absl::get<xla::Shape>(arg.shape), &arg_shape)); in GetTensorShapeFromXlaArgument()
178 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(shape, &resource_update.shape)); in GetOutputInfo()
187 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(shape, &out_desc.shape)); in GetOutputInfo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 451 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(subshape, &shape)); in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 477 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(subshape, &shape)); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 593 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(*subshape, &output_shape)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1453 tensorflow::XLAShapeToTensorShape(arg_shape, &arg_tensor_shape); in SetEntryTupleShapesAndLeafReplication()
|