Home
last modified time | relevance | path

Searched defs:constant_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model_test.cc37 const TensorShape constant_shape({input_height, input_width}); in CreateTestGraph() local
/external/pytorch/torch/csrc/inductor/aoti_runtime/
Dmodel.h355 const int64_t* constant_shape(int64_t idx) const { in constant_shape() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc221 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dquantization_ops_test.cc162 TensorShape constant_shape = TensorShape({channels}); in __anon4f49c2cd0202() local