Home
last modified time | relevance | path

Searched defs:InputShape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dpad_tester.h34 inline PadTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
42 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dsplit_tester.h41 inline SplitTester& InputShape(const std::vector<int32_t>& shape) { in InputShape() function
58 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dreshape_tester.h34 inline ReshapeTester& InputShape(const std::vector<int32_t>& input_shape) { in InputShape() function
43 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dreduce_tester.h36 inline ReduceTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dprelu_tester.h35 inline PreluTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
43 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dquantized_pad_tester.h36 inline QuantizedPadTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
44 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dfully_connected_tester.h36 inline FullyConnectedTester& InputShape( in InputShape() function
46 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dquantized_reduce_tester.h37 inline QuantizedReduceTester& InputShape( in InputShape() function
47 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dquantized_fully_connected_tester.h38 inline QuantizedFullyConnectedTester& InputShape( in InputShape() function
48 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
Dconcatenation_tester.h59 inline std::vector<int32_t> InputShape(size_t i) const { in InputShape() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc79 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape() function in tensorflow::XlaOpKernelContext
83 TensorShape XlaOpKernelContext::InputShape(absl::string_view name) { in InputShape() function in tensorflow::XlaOpKernelContext