Searched refs:GetTensorShape (Results 1 – 15 of 15) sorted by relevance
50 TensorShapeProto GetTensorShape(const HloInstruction* instruction) { in GetTensorShape() function157 *shapes.mutable_list()->add_shape() = GetTensorShape(instruction); in SetNodeAttrs()
45 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
56 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
37 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
58 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
62 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
40 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
189 std::vector<int> GetTensorShape(int index) { in GetTensorShape() function
139 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape()
73 inline nvinfer1::Dims GetTensorShape(const tensorflow::Tensor& tensor) { in GetTensorShape() function1096 GetTensorShape(tensor)); in ConvertConst()1113 weights = ctx.get_temp_weights(dtype, GetTensorShape(tensor)); in ConvertConst()