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

  /external/tensorflow/tensorflow/core/kernels/
sparse_add_op_test.cc 101 DataType val_dtype = tensorflow::DataTypeToEnum<VALTYPE>::value; \
117 if (val_dtype == DT_COMPLEX64) { \
119 } else if (val_dtype == DT_COMPLEX128) { \
134 Tensor expected_values(allocator(), val_dtype, {expected_nnz}); \
159 DataType val_dtype = tensorflow::DataTypeToEnum<VALTYPE>::value; \
178 if (val_dtype == DT_COMPLEX64) { \
180 } else if (val_dtype == DT_COMPLEX128) { \
191 Tensor expected_values(allocator(), val_dtype, TensorShape({0})); \
  /external/tensorflow/tensorflow/core/kernels/data/
sparse_tensor_slice_dataset_op.cc 70 AttrValue val_dtype; variable
71 b->BuildAttrValue(sparse_tensor_.dtype(), &val_dtype);
74 {{"Tvalues", val_dtype}}, output));

Completed in 75 milliseconds