Searched defs:DynamicReshape (Results 1 – 3 of 3) sorted by relevance
978 TEST_F(XlaBuilderTest, DynamicReshape) { in TEST_F() argument
1147 XlaOp XlaBuilder::DynamicReshape(XlaOp operand, in DynamicReshape() function in xla::XlaBuilder4137 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes, in DynamicReshape() function
2679 Status Converter::DynamicReshape(nvinfer1::ITensor* input, in DynamicReshape() function in tensorflow::tensorrt::convert::Converter