Searched refs:compile_time_shapes (Results 1 – 2 of 2) sorted by relevance
197 const std::vector<xla::Shape>& compile_time_shapes) { in UpdateDynamicInputs() argument198 TF_RET_CHECK(runtime_inputs->size() == compile_time_shapes.size()); in UpdateDynamicInputs()199 for (int64 i = 0; i < compile_time_shapes.size(); i++) { in UpdateDynamicInputs()203 if (compile_time_shapes[i].is_static()) { in UpdateDynamicInputs()208 HostShapeToDeviceShape(compile_time_shapes[i]); in UpdateDynamicInputs()
204 const std::vector<xla::ShapeLayout>& compile_time_shapes) { in UpdateDynamicInputs() argument205 TF_RET_CHECK(execution_inputs->size() == compile_time_shapes.size()); in UpdateDynamicInputs()209 for (int64 i = 0; i < compile_time_shapes.size(); i++) { in UpdateDynamicInputs()210 const xla::Shape& compile_time_shape = compile_time_shapes[i].shape(); in UpdateDynamicInputs()