Searched refs:arg_shapes (Results 1 – 4 of 4) sorted by relevance
137 std::vector<const xla::Shape*> arg_shapes; in Compile() local138 arg_shapes.reserve(program_shape->parameters_size()); in Compile()140 arg_shapes.push_back(&program_shape->parameters(i)); in Compile()147 client->Compile(computation, arg_shapes, build_options)); in Compile()
484 std::vector<xla::Shape> arg_shapes(input_mapping->size()); in BuildArguments() local488 XLAShapeForArgument(args[(*input_mapping)[i]], &arg_shapes[i])); in BuildArguments()492 input_shapes->push_back(xla::ShapeUtil::MakeTupleShape(arg_shapes)); in BuildArguments()494 *input_shapes = arg_shapes; in BuildArguments()560 << " shape: " << xla::ShapeUtil::HumanString(arg_shapes[i]) in BuildArguments()
389 tensorflow::gtl::ArraySlice<const Shape*> arg_shapes, in InferConcatOpShape() argument391 if (arg_shapes.empty()) { in InferConcatOpShape()394 if (dimension < 0 || dimension >= ShapeUtil::Rank(*arg_shapes[0])) { in InferConcatOpShape()400 for (const Shape* shape : arg_shapes) { in InferConcatOpShape()443 for (size_t i = 1; i < arg_shapes.size(); ++i) { in InferConcatOpShape()444 new_dimensions[dimension] += arg_shapes[i]->dimensions(dimension); in InferConcatOpShape()1081 tensorflow::gtl::ArraySlice<const Shape*> arg_shapes, in InferMapShape() argument1084 if (arg_shapes.empty()) { in InferMapShape()1089 const Shape* arg_shape = arg_shapes[0]; in InferMapShape()1090 for (size_t i = 1; i < arg_shapes.size(); ++i) { in InferMapShape()[all …]
85 tensorflow::gtl::ArraySlice<const Shape*> arg_shapes,217 tensorflow::gtl::ArraySlice<const Shape*> arg_shapes, int64 dimension);247 tensorflow::gtl::ArraySlice<const Shape*> arg_shapes,