Searched defs:DynamicReshape (Results 1 – 3 of 3) sorted by relevance
1076 TEST_F(XlaBuilderTest, DynamicReshape) { in TEST_F() argument
1214 XlaOp XlaBuilder::DynamicReshape(XlaOp operand, in DynamicReshape() function in xla::XlaBuilder4093 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes, in DynamicReshape() function
2332 Status Converter::DynamicReshape(ITensorProxyPtr input, in DynamicReshape() function in tensorflow::tensorrt::convert::Converter