Searched refs:DynamicReshape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 130 xla::DynamicReshape(ctx->Input(0), output_dim_sizes, in Compile()
172 0, xla::DynamicReshape(ctx->Input(0), output_dim_sizes, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 573 Status DynamicReshape(nvinfer1::ITensor* input,
|
| D | convert_nodes.cc | 2679 Status Converter::DynamicReshape(nvinfer1::ITensor* input, in DynamicReshape() function in tensorflow::tensorrt::convert::Converter
2753 return DynamicReshape(input, slices, params, output, extra_dims, op_instance); in DynamicExpandDims()
2771 return DynamicReshape(input, slices, params, output); in SqueezeTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 478 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes,
1080 friend XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes,
1694 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes,
|
| D | xla_builder_test.cc | 978 TEST_F(XlaBuilderTest, DynamicReshape) { in TEST_F() argument
|
| D | xla_builder.cc | 1147 XlaOp XlaBuilder::DynamicReshape(XlaOp operand, in DynamicReshape() function in xla::XlaBuilder
4137 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes, in DynamicReshape() function
4140 return operand.builder()->DynamicReshape(operand, dim_sizes, new_size_bounds, in DynamicReshape()
|