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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion_test.cc 337 Shape reshape_shape = ShapeUtil::MakeShape(F32, {8, 8}); local
348 HloInstruction::CreateReshape(reshape_shape, broadcast2));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier_test.cc 1983 Shape reshape_shape = builder local
3552 Shape reshape_shape = ShapeUtil::MakeShape(F32, {30, 1, 4}); local
    [all...]
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 601 const string reshape_shape = local
606 CreateMatrixShapeTensorConst(reshape_shape, fc_weights_shape.dims(1), -1,
612 reshape_op->add_input(reshape_shape);
    [all...]

Completed in 173 milliseconds