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

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.h 149 const Shape& init_value_shape, const ProgramShape& scatter_shape);
shape_inference.cc 185 const Shape& init_value_shape,
213 init_value_shape)) {
218 ShapeUtil::HumanString(init_value_shape).c_str());
    [all...]
shape_inference_test.cc 315 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); local
320 matrix_shape, init_value_shape, window, to_apply);
    [all...]

Completed in 563 milliseconds