Searched defs:InputShape (Results 1 – 8 of 8) 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
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
34 inline PreluTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function42 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
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
35 inline FullyConnectedTester& InputShape( in InputShape() function45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
37 inline QuantizedFullyConnectedTester& InputShape( in InputShape() function47 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
77 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape() function in tensorflow::XlaOpKernelContext81 TensorShape XlaOpKernelContext::InputShape(absl::string_view name) { in InputShape() function in tensorflow::XlaOpKernelContext