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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc 621 auto lhs_array = MakeUnique<Array2D<float>>(4, 1); local
622 lhs_array->FillUnique(1.0f);
623 auto lhs_literal = Literal::CreateR2FromArray2D<float>(*lhs_array);
703 auto lhs_array = MakeUnique<Array2D<float>>(4, 3); local
704 lhs_array->FillUnique(1.0f);
705 auto lhs_literal = Literal::CreateR2FromArray2D<float>(*lhs_array);
745 Array3D<float> lhs_array = {{{1, 2, 3}}}; local
746 auto lhs_literal = Literal::CreateR3FromArray3D<float>(lhs_array);
793 Array4D<float> lhs_array(1, 1, 4, 4);
795 lhs_array.FillWithYX(Array2D<float>(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 445 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot); local
454 lhs_array.EmitReadArrayElement(/*index=*/{}, &ir_builder_);
471 result = llvm::ConstantAggregateZero::get(lhs_array.GetElementLlvmType());
506 lhs_array, lhs_reduction_dimension, "lhs", &loop_nest);
532 llvm::Constant::getNullValue(lhs_array.GetElementLlvmType()), // init 0
545 lhs_array.EmitReadArrayElement(lhs_index, &ir_builder_);
    [all...]

Completed in 69 milliseconds