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

  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander_test.cc 45 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); local
55 HloInstruction::CreateParameter(2, offset_shape, "offset"));
58 ShapeUtil::MakeTupleShape({input_shape, scale_shape, offset_shape}),
shape_inference.h 93 const Shape& offset_shape,
100 const Shape& offset_shape, const Shape& mean_shape,
shape_inference.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_batch_normalization.cc 85 const auto& offset_shape = offset_array.shape(); local
87 CHECK(mean_shape.dims() == offset_shape.dims());

Completed in 87 milliseconds