Searched defs:InputShape (Results 1 – 6 of 6) 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 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
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
35 inline FullyConnectedTester& InputShape( in InputShape() function45 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
69 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape() function in tensorflow::XlaOpKernelContext73 TensorShape XlaOpKernelContext::InputShape(absl::string_view name) { in InputShape() function in tensorflow::XlaOpKernelContext