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

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.cc 398 const Shape* arg_shape = nullptr; local
403 if (!arg_shape) {
404 arg_shape = shape;
405 element_type = arg_shape->element_type();
408 if (ShapeUtil::Rank(*arg_shape) != ShapeUtil::Rank(*shape)) {
412 ShapeUtil::Rank(*arg_shape),
413 ShapeUtil::HumanString(*arg_shape).c_str(), ShapeUtil::Rank(*shape),
416 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shape, *shape)) {
419 PrimitiveType_Name(arg_shape->element_type()).c_str(),
423 dimension_number < ShapeUtil::Rank(*arg_shape); ++dimension_number)
1089 const Shape* arg_shape = arg_shapes[0]; local
2437 const Shape& arg_shape = *arg_shapes[i]; local
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 45 const xla::Shape& arg_shape = program_shape.parameters(i); local
47 arg_sizes.push_back(xla::ShapeUtil::ByteSizeOf(arg_shape, kPointerSize));

Completed in 79 milliseconds