| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 116 slice = xla::SetDimensionSize(slice, dynamic_size, i); in Compile()
170 sliced = xla::SetDimensionSize(sliced, dynamic_size, i); in Compile()
|
| D | broadcast_to_op.cc | 43 output = xla::SetDimensionSize( in Compile()
|
| D | sequence_ops.cc | 118 output = xla::SetDimensionSize(output.ValueOrDie(), ctx->Input(1), 0); in Compile()
122 output = xla::SetDimensionSize( in Compile()
|
| D | fill_op.cc | 62 output = xla::SetDimensionSize(output, dynamic_dim_size, i); in Compile()
|
| D | tile_ops.cc | 98 result = xla::SetDimensionSize(result, dynamic_dim_size, i); in Compile()
|
| D | where_op.cc | 83 xla::XlaOp result_padded = xla::SetDimensionSize(result, length, 0); in Compile()
|
| D | shape_op.cc | 136 xla::SetDimensionSize(ctx->Input(0), ctx->Input("size"), dim_index); in Compile()
389 result = xla::SetDimensionSize(result, input_dynamic_dim, i); in Compile()
|
| D | tensor_list_utils.cc | 199 return xla::SetDimensionSize(broadcast, leading_dim_size, 0); in BuildUninitializedTensorList()
284 zeros = xla::SetDimensionSize(zeros, dynamic_dims[i][dim], dim); in CreateZerosTensorListWithShape()
532 read = xla::SetDimensionSize(read, gds, i); in ExecuteTensorListGetItem()
|
| D | categorical_op.cc | 99 log_uniforms = xla::SetDimensionSize(log_uniforms, ctx->Input(1), 1); in Compile()
|
| D | segment_reduction_ops.cc | 111 buffer = xla::SetDimensionSize(buffer, buffer_dims[i], i); in Compile()
|
| D | strided_slice_op.cc | 190 slice = xla::SetDimensionSize(slice, slice_sizes_dynamic[i], i); in EmitDynamicSlice()
340 slice = xla::SetDimensionSize( in Compile()
580 grad = xla::SetDimensionSize(grad, dim_size_scalar, dim); in Compile()
|
| D | unique_op.cc | 211 xla::XlaOp output_dynamic = xla::SetDimensionSize(output, unique_count, 0); in DataOutputFastPath()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 54 iota_s32 = SetDimensionSize(iota_s32, GetDimensionSize(input, i), i); in TopK()
96 iota_s32 = SetDimensionSize(iota_s32, GetDimensionSize(input, i), i); in TopKWithPartitions()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 101 buffer = xla::SetDimensionSize(buffer, buffer_dims[i], i); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 230 auto dx = SetDimensionSize(x, y, 0); in TEST_F()
575 TEST_F(XlaBuilderTest, SetDimensionSize) { in TEST_F() argument
579 auto set_dim_size = SetDimensionSize(p0, p1, 0); in TEST_F()
591 auto set_dim_size = SetDimensionSize(p0, p1, 0); in TEST_F()
|
| D | xla_builder.h | 896 XlaOp SetDimensionSize(XlaOp operand, XlaOp val, int64 dimension);
1429 friend XlaOp SetDimensionSize(XlaOp operand, XlaOp val, int64 dimension);
2542 XlaOp SetDimensionSize(XlaOp operand, XlaOp val, int64 dimension);
|
| D | xla_builder.cc | 3363 XlaOp XlaBuilder::SetDimensionSize(XlaOp operand, XlaOp val, int64 dimension) { in SetDimensionSize() function in xla::XlaBuilder
4974 XlaOp SetDimensionSize(const XlaOp operand, const XlaOp val, int64 dimension) { in SetDimensionSize() function
4975 return operand.builder()->SetDimensionSize(operand, val, dimension); in SetDimensionSize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 266 HLO_MATCHER(SetDimensionSize);
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 61 auto pad_sum = xla::SetDimensionSize(sum, p2, 0); in ReturnDynamicR1()
74 auto pad_sum_dim0 = xla::SetDimensionSize(sum, p2, 0); in ReturnDynamicR2()
75 auto pad_sum_dim1 = xla::SetDimensionSize(pad_sum_dim0, p2, 1); in ReturnDynamicR2()
110 auto pad_sum = xla::SetDimensionSize(sum, p2, 0); in ReturnDynamicR1Tuple()
111 auto pad_sub = xla::SetDimensionSize(sub, p2 + one, 0); in ReturnDynamicR1Tuple()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 1120 arg_handles[i] = xla::SetDimensionSize( in BuildArguments()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.td | 1187 string summary = "SetDimensionSize operator";
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1658 ## SetDimensionSize section in Operation Semantics
1661 [`XlaBuilder::SetDimensionSize`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla…
1666 <b> `SetDimensionSize(operand, size, dimension)` </b>
|