Searched refs:DimensionSizes (Results 1 – 3 of 3) sorted by relevance
172 std::vector<int64> DimensionSizes() const;
406 std::vector<int64> XlaCompiler::Argument::DimensionSizes() const { in DimensionSizes() function in tensorflow::XlaCompiler::Argument876 xla::Reshape(arg_handles[i], arg.DimensionSizes()), arg.type); in BuildArguments()
129 signature.arg_shapes.emplace_back(arg.type, arg.DimensionSizes()); in BuildSignature()