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

  /external/tensorflow/tensorflow/core/ops/
array_ops_test.cc 550 Tensor dim_t; local
551 op.input_tensors[1] = &dim_t;
555 dim_t = test::AsScalar<int32>(idx);
562 dim_t = test::AsScalar<int32>(idx);
567 dim_t = test::AsScalar<int64>(idx);
572 dim_t = test::AsScalar<int32>(idx);
577 dim_t = test::AsScalar<int64>(idx);
584 dim_t = test::AsScalar<int32>(idx);
589 dim_t = test::AsScalar<int64>(idx);
595 dim_t = test::AsScalar<int32>(idx)
    [all...]
math_ops.cc 797 const Tensor* dim_t = c->input_tensor(1); local
798 if (dim_t == nullptr) {
812 if (dim_t->dtype() == DT_INT32) {
813 dimension_val = dim_t->scalar<int32>()();
815 dimension_val = dim_t->scalar<int64>()();
    [all...]
array_ops.cc     [all...]

Completed in 2639 milliseconds