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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
select_op.cc 74 std::vector<int64> dim_order(then_shape.dims());
75 dim_order[0] = then_shape.dims() - 1;
76 std::iota(dim_order.begin() + 1, dim_order.end(), 0);
77 cond_handle = builder->Transpose(cond_handle, dim_order);
  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier_test.cc 1788 const char* dim_order = "NHWC"; \/\/ can use chars NHWC in any order. member in struct:ConvTestOptions
    [all...]

Completed in 75 milliseconds