HomeSort by relevance Sort by last modified time
    Searched refs:argument_shapes (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc 266 tensorflow::gtl::ArraySlice<const Shape*> argument_shapes,
272 if (program_shape.parameters_size() != argument_shapes.size()) {
275 argument_shapes.size());
277 for (int i = 0; i < argument_shapes.size(); ++i) {
280 if (!ShapeUtil::Compatible(*argument_shapes[i],
287 ShapeUtil::HumanString(*argument_shapes[i]).c_str());
291 *argument_shapes[i]));
329 std::vector<const Shape*> argument_shapes; local
331 argument_shapes.push_back(&arg->on_host_shape());
333 return CreateModuleConfig(program_shape, argument_shapes, &execution_options
    [all...]
service.h 278 tensorflow::gtl::ArraySlice<const Shape*> argument_shapes,
shape_inference.cc 2423 string argument_shapes = local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 258 const std::vector<Shape>& argument_shapes,
261 argument_shape_pointers.reserve(argument_shapes.size());
262 for (auto& argument_shape : argument_shapes) {
local_computation_builder.h 107 const std::vector<Shape>& argument_shapes,
xla_client.py 369 def Compile(self, argument_shapes=(), compile_options=None, layout_fn=None):
380 argument_shapes: parameter shapes -- they are first laid out by layout_fn
394 argument_shapes = [
395 shape.map_leaves(layout_fn) for shape in argument_shapes
402 self.c_local_computation.Compile(argument_shapes, compile_options),
410 argument_shapes=[Shape.from_numpy(arg) for arg in arguments],
418 argument_shapes = [Shape.from_numpy(arg) for arg in arguments]
420 argument_shapes = [
421 shape.map_leaves(layout_fn) for shape in argument_shapes
424 argument_shapes = [None for shape in argument_shapes
    [all...]

Completed in 97 milliseconds