Searched defs:InputShape (Results 1 – 13 of 13) sorted by relevance
34 inline PadTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function42 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
41 inline SplitTester& InputShape(const std::vector<int32_t>& shape) { in InputShape() function58 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
34 inline ReshapeTester& InputShape(const std::vector<int32_t>& input_shape) { in InputShape() function43 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
36 inline ReduceTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
35 inline PreluTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function43 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
36 inline QuantizedPadTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function44 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
36 inline FullyConnectedTester& InputShape( in InputShape() function46 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
37 inline QuantizedReduceTester& InputShape( in InputShape() function47 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
38 inline QuantizedFullyConnectedTester& InputShape( in InputShape() function48 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
59 inline std::vector<int32_t> InputShape(size_t i) const { in InputShape() function
79 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape() function in tensorflow::XlaOpKernelContext83 TensorShape XlaOpKernelContext::InputShape(absl::string_view name) { in InputShape() function in tensorflow::XlaOpKernelContext
28 "InputShape": { object
25 InputShape, enumerator